package com.zj.sixtyDayChallenge.backtracking;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * @program: algorithm
 * @description: 组合
 * 77 combinations
 * //给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
 * //
 * // 你可以按 任何顺序 返回答案。
 * //
 * //
 * //
 * // 示例 1：
 * //
 * //
 * //输入：n = 4, k = 2
 * //输出：
 * //[
 * //  [2,4],
 * //  [3,4],
 * //  [2,3],
 * //  [1,2],
 * //  [1,3],
 * //  [1,4],
 * //]
 * //
 * // 示例 2：
 * //
 * //
 * //输入：n = 1, k = 1
 * //输出：[[1]]
 * //
 * //
 * //
 * // 提示：
 * //
 * //
 * // 1 <= n <= 20
 * // 1 <= k <= n
 * //
 * //
 * // Related Topics 回溯 👍 1348 👎 0
 * @author: Zhang Bo
 * @create: 2022-01-14 17:05
 **/
public class Combinations {
    public static void main(String[] args) {
        Solution solution = new Combinations().new Solution();
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        List<List<Integer>> ret = new ArrayList<>();
        LinkedList<Integer> path = new LinkedList<>();

        public List<List<Integer>> combine(int n, int k) {
            helper(n, k, 1);
            return ret;
        }

        public void helper(int n, int k, int startIndex) {
            //终止条件
            if (path.size() == k) {
                ret.add(new ArrayList<>(path));
            }

            //如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了，那么就没有必要搜索了。
            //所以可以优化为 for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) 这种形式
            for (int i = startIndex; i <= n; i++) {
                path.add(i);
                helper(n, k, i + 1);
                path.removeLast();
            }

        }


        public List<List<Integer>> combine02(int n, int k) {
            helper02(n, k, 1);
            return ret;
        }


        public void helper02(int n, int k, int startIndex) {
            if (path.size() == k) {
                ret.add(path);
            }

            for (int i = startIndex; i <= n; i++) {
                path.add(i);
                helper02(n, k, i + 1);
                path.removeLast();
            }
        }

    }
//leetcode submit region end(Prohibit modification and deletion)

}
